package com.hc.programming.tree;

import java.util.*;

/**
 * 给你二叉树的根节点 root ，返回它节点值的 前序 遍历。
 * <p>
 * 示例 1：
 * 输入：root = [1,null,2,3]
 * 输出：[1,2,3]
 * 解释：
 * <a href="./二叉树的前序遍历-示例1.png">示例1</a>
 * <p>
 * 示例 2：
 * 输入：root = [1,2,3,4,5,null,8,null,null,6,7,9]
 * 输出：[1,2,4,5,6,7,3,8,9]
 * 解释：
 * <a href="./二叉树的前序遍历-示例2.png">示例2</a>
 * <p>
 * 示例 3：
 * 输入：root = []
 * 输出：[]
 * 示例 4：
 * 输入：root = [1]
 * 输出：[1]
 * <p>
 * 提示：
 * 树中节点数目在范围 [0, 100] 内
 * -100 <= Node.val <= 100
 * <p>
 * 进阶：递归算法很简单，你可以通过迭代算法完成吗？
 *
 * @author huangchao E-mail:fengquan8866@163.com
 * @version 创建时间：2024/9/5 11:24
 */
public class 二叉树的前序遍历 {
    public static void main(String[] args) {
        Integer[] arr = new Integer[]{1, 2, 3, 4, 5, null, 8, null, null, 6, 7, 9};
        TreeNode tree = TreeNode.tree(arr);
        System.out.println(tree);
        System.out.println(Arrays.toString(arr) + "=[1, 2, 4, 5, 6, 7, 3, 8, 9],--" + preorderTraversal(tree));
    }

    public static List<Integer> preorderTraversal(TreeNode root) {
//        return 递归(root);
        return 迭代(root);
    }

    private static List<Integer> 迭代(TreeNode root) {
        List<Integer> rst = new ArrayList<>();
        if (root == null) return rst;
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode node = stack.pop();
            rst.add(node.val);
            if (node.right != null) stack.push(node.right);
            if (node.left != null) stack.push(node.left);
        }
        return rst;
    }

    public static List<Integer> 递归(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        preorderTraversal(root, list);
        return list;
    }

    private static void preorderTraversal(TreeNode root, List<Integer> list) {
        if (root == null) return;
        list.add(root.val);
        preorderTraversal(root.left, list);
        preorderTraversal(root.right, list);
    }
}
